package com.example.demo.web.controller;

import com.example.demo.bean.Dataset;
import com.example.demo.service.IDatasetService;
import com.example.demo.utils.ExcelUtils;
import com.example.demo.utils.Message;
import com.example.demo.utils.MessageUtil;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;

import javax.servlet.http.HttpServletResponse;
import javax.validation.constraints.NotNull;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@RestController
@RequestMapping("/dataset")
public class DatasetController {

    @Autowired
    private IDatasetService datasetService;
    @ApiOperation(value = "导入到Excel中", notes = "注意！测试的时候请将地址粘贴到浏览器地址栏测试", produces = "application/octet-stream")
    @GetMapping("download")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "name", value = "车辆数据",paramType = "query"),
    })
    public void download(HttpServletResponse response, String name) throws Exception {
    //查询数据
            List<Dataset> list = datasetService.query(name);
    // 2. 将excel导出
    // 2.1 excel头部数据
    String[] headList = new String[] { "id", "名称", "介绍", "照片"};
    // 2.2 属性
    String[] fieldList = new String[] { "id", "name", "introduce", "photo"};
    // 2.3 数据列，需要转化
    List<Map<String, Object>> dataList = new ArrayList<>();

        for (Dataset u : list) {
        // 每一行map相当于一行数据，一行数据中多列
        Map<String, Object> map = new HashMap<>();
//        map.put("id", u.getId());
        map.put("name", u.getName());
        map.put("introduce", u.getIntroduce());
        map.put("photo", u.getPhoto());

        // dataList表示所有行
        dataList.add(map);
    }
    // 2.4 导出文件名
    String excelName = "datasets";
    // 3. 调用工具方法导出Excel
        ExcelUtils.createExcel(response, excelName, headList, fieldList, dataList);

}
    @GetMapping("query")
    public Message query(String name) {
        List<Dataset> list = datasetService.query(name);
        return MessageUtil.success(list);
    }

    @PostMapping("saveOrUpdate")
    public Message saveOrUpdate(Dataset dataset) {
        datasetService.saveOrUpdate(dataset);
        return MessageUtil.success("操作成功");
    }

    @DeleteMapping("deleteById")

    @ApiImplicitParams({
            @ApiImplicitParam(name = "id", value = "应用编号", required = true, paramType = "query"),
    })
    public Message deleteById(@NotNull Long id) {
        datasetService.deleteById(id);
        return MessageUtil.success("删除成功");
    }

}
